package primetest.faces;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;


import primetest.dto.BedDto;

public class BedListModel extends ListDataModel<BedDto> implements SelectableDataModel<BedDto> {

	public BedListModel(List<BedDto> bedList){
		super(bedList);
	}

	@Override
	public BedDto getRowData(String rowKey) {
		@SuppressWarnings("unchecked")
		List<BedDto> bedList = (List<BedDto>)getWrappedData();
		for (BedDto bedDto : bedList) {
			if(bedDto.getKey().equals(rowKey)){
				return bedDto;
			}
		}
		return null;
	}

	@Override
	public Object getRowKey(BedDto bedDto) {
		return bedDto.getKey();
	}

}
